How to know size of a dataframe in PySpark